17905f56c9c697dd3b0c34cfbc184adb3d643af8,modules/repository-datastax/src/test/java/com/intuit/wasabi/repository/cassandra/impl/CassandraAssignmentsRepositoryTest.java,CassandraAssignmentsRepositoryTest,testGetAssignmentsMultiple,#,245
Before Change
Experiment.withID(Experiment.ID.valueOf(random)).build()
);
for (ExperimentUserByUserIdContextAppNameExperimentId item : mocked) {
experimentTable.put(Experiment.ID.valueOf(item.getExperimentId()),
Experiment.Label.valueOf("test-" + item.getBucket()),
Experiment.withID(Experiment.ID.valueOf(item.getExperimentId()))
.withLabel(Experiment.Label.valueOf("test-" + item.getBucket())).build()
);
}
doReturn(mocked.stream()).when(spyRepository).getUserIndexStream(anyString(), anyString(), anyString());
After Change
@Test
public void testGetAssignmentsMultiple() {
Experiment.ID expId1 = Experiment.ID.newInstance();
Experiment.ID expId2 = Experiment.ID.newInstance();
Date endTime = new Date(System.currentTimeMillis() + 30 * 24 * 60 * 60 * 1000);
Experiment exp1 = Experiment.withID(expId1).withEndTime(endTime)
.withLabel(Experiment.Label.valueOf("Exp1")).build();
Experiment exp2 = Experiment.withID(expId2).withEndTime(endTime)
.withLabel(Experiment.Label.valueOf("Exp2")).build();
List<ExperimentUserByUserIdContextAppNameExperimentId> mocked = new ArrayList<>();
mocked.add(ExperimentUserByUserIdContextAppNameExperimentId.builder()
.appName(APPLICATION_NAME.toString())
.experimentId(expId1.getRawID())
.context("test")
.bucket("bucket1")
.build()
);
mocked.add(ExperimentUserByUserIdContextAppNameExperimentId.builder()
.appName(APPLICATION_NAME.toString())
.experimentId(expId2.getRawID())
.context("test")
.bucket("bucket2")
.build()
);
Map<Experiment.ID, Experiment> experimentMap = newHashMap();
experimentMap.put(expId1, exp1);
experimentMap.put(expId2, exp2);
doReturn(mocked.stream()).when(spyRepository).getUserIndexStream(anyString(), anyString(), anyString());